package com.xzc.servlet;

import org.apache.commons.io.IOUtils;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;

/**
 * @author 徐州城
 * @version 1.0
 */
public class Download extends HttpServlet {

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1、获取下载的文件名
        String downloadFile = "2.jpg";
        //2、读取要下载的文件内容（通过ServletContext对象可以读取）
        ServletContext servletContext = getServletContext();

        //获取要下载的文件类型
        String mimeType = servletContext.getMimeType("/file/" + downloadFile);
        //4、在回传前，通过响应头告诉客户端返回的数据类型
        resp.setContentType(mimeType);

        //5、还要告诉客户端收到的数据是用于下载使用（还是使用响应头）
        // Contend-Disposition响应头，表示收到的数据怎么处理
        // attachment表示附件，表示下载使用
        // filename= 表示指定下载的文件名
        //url编码是把汉字转换为 %xx%xx的格式
        resp.setHeader("Contend-Disposition","attachment; filename=" + URLEncoder.encode("中国.jpg","UTF-8"));

        //3、把下载的文件回传给客户端
        InputStream resourceAsStream = servletContext.getResourceAsStream("/file/" + downloadFile);
        //获取响应的输出流
        OutputStream outputStream = resp.getOutputStream();
        //读取输入流中的全部的数据，复制给 输出流，输出给客户端
        IOUtils.copy(resourceAsStream,outputStream);
    }
}
